package com.tree.leetcode.array;

import cn.hutool.json.JSON;
import cn.hutool.json.JSONUtil;
import com.tree.sort.BinarySearch;

/**
 * @ClassName com.tree.leetcode.array
 * Description: 34. 在排序数组中查找元素的第一个和最后一个位置
给你一个按照非递减顺序排列的整数数组 nums，和一个目标值 target。请你找出给定目标值在数组中的开始位置和结束位置。
如果数组中不存在目标值 target，返回 [-1, -1]。
你必须设计并实现时间复杂度为 O(log n) 的算法解决此问题。
示例 1：
输入：nums = [5,7,7,8,8,10], target = 8
输出：[3,4]
示例 2：
输入：nums = [5,7,7,8,8,10], target = 6
输出：[-1,-1]
示例 3：
输入：nums = [], target = 0
输出：[-1,-1]

 * @Author frankzsliu
 * @Date 2024/2/4 23:17
 * @Version 1.0
 */
public class SearchRange_34 {
    public static void main(String[] args) {
        int[] nums = {5,7,7,8,8,10};
        int target = 8;
        System.out.println(JSONUtil.toJsonStr(searchRange(nums,target)));
    }


    public static int[] searchRange(int[] nums, int target) {
        //寻找左边界(这里寻找第一个 >= target的索引)
        int leftIndex = search2(nums, target);
        if(leftIndex >= nums.length || nums[leftIndex] != target){
            return new int[]{-1,-1};
        }
        //寻找右边界(这里寻找第一个 >= target+1的索引)
        //找到大于等于target+1的钱一个位置
        int rightIndex = search2(nums, target+1);
        return new int[]{leftIndex,rightIndex-1};
    }

    /**
     * 寻找第一个>=目标值的索引, 找不到则返回数组长度
     */
    private static int search(int[] nums, int target) {
        //初始条件，设置左边界和右边界
        int left = 0;
        int right = nums.length - 1;
        while (left <= right){
            int mid = (right - left) / 2 + left;
            //大于等于target
            if (nums[mid] >= target){
                //从mid-1开始找
                right = mid - 1;
            } else {
                left = mid + 1;
            }
        }
        //找大于等于的数，返回left
        return left;
    }

    private static int search2(int[] nums, int target) {
        //初始条件，设置左边界和右边界
        int left = 0;
        int right = nums.length - 1;
        while (left <= right){
            int mid = (left+right)/2; //(right - left) / 2 + left;
            if(nums[mid] > target) right = mid -1 ;
            else if(nums[mid] < target) left = mid+1;
            else {
                //当有多个同样的数字的时候， 直接返回mid就有可能错误， 应当返回第一个
                if(nums[left] == nums[mid]) return left;
                else return mid;
            }
        }
        //找大于等于的数，返回left最左边的界限
        return left;
    }

}
